/**
 *    Copyright 2009-2020 the original author or authors.
 *
 *    Licensed under the Apache License, Version 2.0 (the "License");
 *    you may not use this file except in compliance with the License.
 *    You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 *    Unless required by applicable law or agreed to in writing, software
 *    distributed under the License is distributed on an "AS IS" BASIS,
 *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *    See the License for the specific language governing permissions and
 *    limitations under the License.
 */
package org.apache.ibatis.aapl.script.creator;


import org.apache.ibatis.aapl.annotation.TableWatcher;
import org.apache.ibatis.aapl.util.Clause;
import org.apache.ibatis.aapl.util.NameUtil;

import java.util.List;

public class UpdateSqlCreator extends UpdateAllSqlCreator {
  @Override
  protected String buildSql(List<String> fields, TableWatcher watcher, Clause clause) {
    StringBuilder sb = new StringBuilder();
    StringBuilder begin = getStringBuilder(watcher);
    sb.append("<script>");
    sb.append(begin);
    List<String> excepts = clause.except();
    sb.append("<trim suffixOverrides=','>\n");
    for (int i = 0; i < fields.size(); i++) {
      String str = fields.get(i);
      if (excepts.contains(str)) {
        continue;
      }
      if (!"id".equals(str)) {
        StringBuilder ifNode = getIfNode(str);
        sb.append(ifNode);
      }
    }
    sb.append("</trim> where id = #{id} </script>");
    return sb.toString();
  }

  private StringBuilder getIfNode(String str) {
    StringBuilder sb = new StringBuilder();
    sb.append("<if test=\"");
    sb.append(str + "!= null and ");
    sb.append(str + " != ''\">\n");
    sb.append(NameUtil.camel2Underline(str));
    sb.append(" = ");
    sb.append("#{");
    sb.append(str);
    sb.append("},");
    sb.append("</if>\n");
    return sb;
  }
}
